package com.suning.config;

import com.suning.mapper.EbProductCategoryMapper;
import com.suning.model.EbProductCategory;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import javax.servlet.ServletContext;
import java.util.List;

@Component
public class InitContext implements CommandLineRunner {
    @Resource
    private EbProductCategoryMapper ebProductCategoryMapper;

    @Resource
    private ServletContext application;

    @Override
    public void run(String... args) throws Exception {
        System.out.println("=====spring boot项目自定义初始化=====");

        List<EbProductCategory> bigs = ebProductCategoryMapper.selectBigCategory();
        List<EbProductCategory> smalls = ebProductCategoryMapper.selectSmallCategory();

        application.setAttribute("bigs",bigs);
        application.setAttribute("smalls",smalls);
    }
}
